Skip to content

【25-Q4-生态建设】模型迁移-研发效能部-模型训练-在PyTorch框架上支持 VovNet V2 and V1在Cifar100上的训练#393

Open
x0212wwl wants to merge 4 commits intoTecorigin:mainfrom
x0212wwl:VovNet
Open

【25-Q4-生态建设】模型迁移-研发效能部-模型训练-在PyTorch框架上支持 VovNet V2 and V1在Cifar100上的训练#393
x0212wwl wants to merge 4 commits intoTecorigin:mainfrom
x0212wwl:VovNet

Conversation

@x0212wwl
Copy link

● 当前软件栈版本:
image

● 源码参考链接:https://github.com/huggingface/pytorch-image-models
● commit id:x0212wwl@ https://github.com/x0212wwl
● 工作目录:PyTorch/build-in/classification/VovNet/
● 训练内容:使用1张TECO_AICARD_01芯片,在PyTorch框架上支持VovNet在Cifar100数据集上的训练。
● 运行脚本如下:
SDAA_VISIBLE_DEVICES=8,9,10,11 weloTrain.py --arch vovnetv1 --dataset cifar100 --datapath ./data --batch_size 32 --epochs 100 | tee vovnetv1Cifar100Sdaa.log
SDAA_VISIBLE_DEVICES=8,9,10,11 weloTrain.py --arch vovnetv2 --dataset cifar100 --datapath ./data --batch_size 32 --epochs 100 | tee vovnetv2Cifar100Sdaa.log

● 100iters损失:

  • VovNet V1
image

MeanRelativeError: -0.057127017790560425
MeanAbsoluteError: -0.7131010000000001
Rule,mean_absolute_error -0.7131010000000001
pass mean_relative_error=-0.057127017790560425 <= 0.05 or mean_absolute_error=-0.7131010000000001 <= 0.0002

  • VovNet V2
image MeanRelativeError: 0.005383627924133852 MeanAbsoluteError: 0.023844999999999974 Rule,mean_relative_error 0.005383627924133852 pass mean_relative_error=0.005383627924133852 <= 0.05 or mean_absolute_error=0.023844999999999974 <= 0.0002

@x0212wwl x0212wwl changed the title 元碁智汇·定义训练未来-北邮&苏大-模型-在PyTorch框架上支持VovNet V2 and V1在Cifar100上的训练 【25-Q4-生态建设】模型迁移-研发效能部-模型训练-在PyTorch框架上支持 VovNet V2 and V1在Cifar100上的训练 Dec 24, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant